package com.zenchn.utils;

/**
 * PwdCheckStrength：
 * 检测密码强度
 * @author dingxinfa
 * @date 2020-05-06 09:29
 */
public class CheckStrength {

    public enum LEVEL {
        EASY, MIDIUM, STRONG, VERY_STRONG, EXTREMELY_STRONG
    }
    /**
     * NUM 数字
     * SMALL_LETTER 小写字母
     * CAPITAL_LETTER 大写字母
     * OTHER_CHAR 特殊字符
     */
    private static final int NUM = 1;
    private static final int SMALL_LETTER = 2;
    private static final int CAPITAL_LETTER = 3;
    private static final int OTHER_CHAR = 4;
    /**
     * 简单的密码字典
     */
    private final static String[] DICTIONARY = {"password", "abc123", "iloveyou", "adobe123", "123123", "sunshine",
            "1314520", "a1b2c3", "123qwe", "aaa111", "qweasd", "admin", "passwd"};
    /**
     *检查字符类型，包括num、大写字母、小写字母和其他字符。
     *
     * @param c
     * @return
     */
    private static int checkCharacterType(char c) {
        if (c >= 48 && c <= 57) {
            return NUM;
        }
        if (c >= 65 && c <= 90) {
            return CAPITAL_LETTER;
        }
        if (c >= 97 && c <= 122) {
            return SMALL_LETTER;
        }
        return OTHER_CHAR;
    }
    /**
     * 按不同类型计算密码的数量
     *
     * @param password
     * @param type
     * @return
     */
    private static int countLetter(String password, int type) {
        int count = 0;
        if (null != password && password.length() > 0) {
            for (char c : password.toCharArray()) {
                if (checkCharacterType(c) == type) {
                    count++;
                }
            }
        }
        return count;
    }
    /**
     * 检查密码的强度
     *
     * @param password
     * @return strength level
     */
    public static int checkPasswordStrength(String password) {
        if (StringUtils.equalsNull(password)) {
            throw new IllegalArgumentException("password is empty");
        }
        int len = password.length();
        int level = 0;
        // 增加点
        //判断密码是否含有数字有level++
        if (countLetter(password, NUM) > 0) {
            level++;
        }
        //判断密码是否含有小写字母有level++
        if (countLetter(password, SMALL_LETTER) > 0) {
            level++;
        }
        //判断密码是否还有大写字母有level++
        if (len > 4 && countLetter(password, CAPITAL_LETTER) > 0) {
            level++;
        }
        //判断密码是否还有特殊字符有level++
        if (len > 6 && countLetter(password, OTHER_CHAR) > 0) {
            level++;
        }
        //密码长度大于4并且2种类型组合......（不一一概述）
        if (len > 4 && countLetter(password, NUM) > 0 && countLetter(password, SMALL_LETTER) > 0
                || countLetter(password, NUM) > 0 && countLetter(password, CAPITAL_LETTER) > 0
                || countLetter(password, NUM) > 0 && countLetter(password, OTHER_CHAR) > 0
                || countLetter(password, SMALL_LETTER) > 0 && countLetter(password, CAPITAL_LETTER) > 0
                || countLetter(password, SMALL_LETTER) > 0 && countLetter(password, OTHER_CHAR) > 0
                || countLetter(password, CAPITAL_LETTER) > 0 && countLetter(password, OTHER_CHAR) > 0) {
            level++;
        }
        //密码长度大于6并且3中类型组合......（不一一概述）
        if (len > 6 && countLetter(password, NUM) > 0 && countLetter(password, SMALL_LETTER) > 0
                && countLetter(password, CAPITAL_LETTER) > 0 || countLetter(password, NUM) > 0
                && countLetter(password, SMALL_LETTER) > 0 && countLetter(password, OTHER_CHAR) > 0
                || countLetter(password, NUM) > 0 && countLetter(password, CAPITAL_LETTER) > 0
                && countLetter(password, OTHER_CHAR) > 0 || countLetter(password, SMALL_LETTER) > 0
                && countLetter(password, CAPITAL_LETTER) > 0 && countLetter(password, OTHER_CHAR) > 0) {
            level++;
        }
        //密码长度大于8并且4种类型组合......（不一一概述）
        if (len > 8 && countLetter(password, NUM) > 0 && countLetter(password, SMALL_LETTER) > 0
                && countLetter(password, CAPITAL_LETTER) > 0 && countLetter(password, OTHER_CHAR) > 0) {
            level++;
        }
        //密码长度大于6并且2种类型组合每种类型长度大于等于3或者2......（不一一概述）
        if (len > 6 && countLetter(password, NUM) >= 3 && countLetter(password, SMALL_LETTER) >= 3
                || countLetter(password, NUM) >= 3 && countLetter(password, CAPITAL_LETTER) >= 3
                || countLetter(password, NUM) >= 3 && countLetter(password, OTHER_CHAR) >= 2
                || countLetter(password, SMALL_LETTER) >= 3 && countLetter(password, CAPITAL_LETTER) >= 3
                || countLetter(password, SMALL_LETTER) >= 3 && countLetter(password, OTHER_CHAR) >= 2
                || countLetter(password, CAPITAL_LETTER) >= 3 && countLetter(password, OTHER_CHAR) >= 2) {
            level++;
        }
        //密码长度大于8并且3种类型组合每种类型长度大于等于3或者2......（不一一概述）
        if (len > 8 && countLetter(password, NUM) >= 2 && countLetter(password, SMALL_LETTER) >= 2
                && countLetter(password, CAPITAL_LETTER) >= 2 || countLetter(password, NUM) >= 2
                && countLetter(password, SMALL_LETTER) >= 2 && countLetter(password, OTHER_CHAR) >= 2
                || countLetter(password, NUM) >= 2 && countLetter(password, CAPITAL_LETTER) >= 2
                && countLetter(password, OTHER_CHAR) >= 2 || countLetter(password, SMALL_LETTER) >= 2
                && countLetter(password, CAPITAL_LETTER) >= 2 && countLetter(password, OTHER_CHAR) >= 2) {
            level++;
        }
        //密码长度大于10并且4种类型组合每种类型长度大于等于2......（不一一概述）
        if (len > 10 && countLetter(password, NUM) >= 2 && countLetter(password, SMALL_LETTER) >= 2
                && countLetter(password, CAPITAL_LETTER) >= 2 && countLetter(password, OTHER_CHAR) >= 2) {
            level++;
        }
        //特殊字符>=3 level++;
        if (countLetter(password, OTHER_CHAR) >= 3) {
            level++;
        }
        //特殊字符>=6 level++;
        if (countLetter(password, OTHER_CHAR) >= 6) {
            level++;
        }
        //长度>12 >16 level++
        if (len > 12) {
            level++;
            if (len >= 16) {
                level++;
            }
        }
        // 减少点
        if ("abcdefghijklmnopqrstuvwxyz".contains(password) || "ABCDEFGHIJKLMNOPQRSTUVWXYZ".contains(password)) {
            level--;
        }
        if ("qwertyuiop".contains(password) || "asdfghjkl".contains(password) || "zxcvbnm".contains(password)) {
            level--;
        }
        if (StringUtils.isNumeric(password) && ("01234567890".contains(password) || "09876543210".contains(password))) {
            level--;
        }
        if (countLetter(password, NUM) == len || countLetter(password, SMALL_LETTER) == len
                || countLetter(password, CAPITAL_LETTER) == len) {
            level--;
        }
        // aaabbb
        if (len % 2 == 0) {
            String part1 = password.substring(0, len / 2);
            String part2 = password.substring(len / 2);
            if (part1.equals(part2)) {
                level--;
            }
            if (StringUtils.isCharEqual(part1) && StringUtils.isCharEqual(part2)) {
                level--;
            }
        }
        // ababab
        if (len % 3 == 0) {
            String part1 = password.substring(0, len / 3);
            String part2 = password.substring(len / 3, len / 3 * 2);
            String part3 = password.substring(len / 3 * 2);
            if (part1.equals(part2) && part2.equals(part3)) {
                level--;
            }
        }
        // 19881010 or 881010
        if (StringUtils.isNumeric(password) && len >= 6) {
            int year = 0;
            if (len == 8 || len == 6) {
                year = Integer.parseInt(password.substring(0, len - 4));
            }
            int size = StringUtils.sizeOfInt(year);
            int month = Integer.parseInt(password.substring(size, size + 2));
            int day = Integer.parseInt(password.substring(size + 2, len));
            if (year >= 1950 && year < 2050 && month >= 1 && month <= 12 && day >= 1 && day <= 31) {
                level--;
            }
        }
        // dictionary
        if (null != DICTIONARY) {
            for (String s : DICTIONARY) {
                if (password.equals(s) || s.contains(password)) {
                    level--;
                    break;
                }
            }
        }
        if (len < 6) {
            level--;
            if (len < 4) {
                level--;
                if (len < 3) {
                    level = 0;
                }
            }
        }
        if (StringUtils.isCharEqual(password)) {
            level = 0;
        }
        if (level < 0) {
            level = 0;
        }
        return level;
    }
    /**
     *获得密码强度等级，包括简单、复杂、强、强、强
     *
     * @param password
     * @return
     */
    public static LEVEL getPasswordLevel(String password) {
        int level = checkPasswordStrength(password);
        switch (level) {
            case 0:
            case 1:
            case 2:
            case 3:
                return LEVEL.EASY;
            case 4:
            case 5:
            case 6:
                return LEVEL.MIDIUM;
            case 7:
            case 8:
            case 9:
                return LEVEL.STRONG;
            case 10:
            case 11:
            case 12:
                return LEVEL.VERY_STRONG;
            default:
                return LEVEL.EXTREMELY_STRONG;
        }
    }
    public static void main(String[] args) {

        String pwd="1全占我";
        System.out.println(checkPasswordStrength(pwd));

        System.out.println(getPasswordLevel(pwd));
    }
}
